package com.treeroot.simple;

import java.util.*;
import java.util.stream.Stream;

/**
 * Create By TreeRoot
 * Create Time: 2024/11/09 08:58
 * Class Name: SumOfTwoNumbers
 * Description:
 * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
 * <p>
 * 你可以假设每种输入只会对应一个答案，并且你不能使用两次相同的元素。
 * <p>
 * 你可以按任意顺序返回答案。
 *
 * @author TreeRoot
 */
public class SumOfTwoNumbers {
    /**
     * 两数之和 时间复杂度最佳O(n)
     * @param nums  数组
     * @param target 目标值
     * @return 两数之和的下标
     */
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i];
            // 如果map中存在complement，返回map中complement的value和当前下标
            if (map.containsKey(complement)) {
                return new int[]{map.get(complement), i};
            }
            //将当前值作为key，下标作为value存入map
            map.put(nums[i], i);
        }
        return null;
    }
}
